package org.example.pojo;

import lombok.Data;

import java.util.ArrayList;
import java.util.List;

/**
 * ChatGPT请求参数对象，参考文档：https://platform.openai.com/docs/api-reference/chat/create
 *
 * {
 *     "model":"gpt-3.5-turbo",
 *     "messages":[
 *         {
 *             "role":"user",
 *             "content":"SpringBoot是什么"
 *         }
 *     ]
 * }
 */
//构建请求的参数对象
@Data
public class ChatMessageParam {
    //model 代表了使用chatgpt的哪个模型
    private String model = "gpt-3.5-turbo";
    //请求消息，要去以数组格式
    private List<ChatMessage> messages = new ArrayList<>();
    //往message中添加message
    public void addMessages(ChatMessage message) {
        this.messages.add(message);
    }

    public ChatMessageParam(){}

    public ChatMessageParam(ChatMessage message){
        this.messages.add(message);
    }
}
